using JADE, JuMP 
 
# Choose your solver 
using Gurobi 
env = Gurobi.Env() 
optimizer = optimizer_with_attributes(() -> Gurobi.Optimizer(env), "OutputFlag" => 0) 
 
## Set directory containing the Input / Output subdirectories 
cd(@__DIR__) 
ENV["JADE_DIR"] = dirname(dirname(@__DIR__)) 
 
## Modify these settings 
inputdir = basename(@__DIR__) 
runfile = "run" 
training = true 
simulation = true 
 
## Set up inputs for JADE models 
rundata = define_JADE_model(inputdir, run_file = runfile) 
 
## Create JADE model from the runfile data 
model = create_JADE_model(rundata, optimizer) 
 
if training 
    solve_options = define_JADE_solve_options(inputdir, run_file = runfile) 
    optimize_policy!(model, solve_options) 
end 
 
if simulation 
    sim_settings = define_JADE_simulation(inputdir, run_file = runfile) 
    results = simulate(model, sim_settings) 
end